package com.xpgk.mvc.statistic.measure;

import java.io.IOException;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.net.URLEncoder;
import java.util.ArrayList;
import java.util.List;

import com.jfinal.kit.LogKit;
import com.jfinal.plugin.activerecord.Db;
import com.jfinal.plugin.activerecord.Record;
import com.jfinal.render.Render;
import com.jfinal.render.RenderException;
import com.xpgk.mvc.codeSet.Code;
import com.xpgk.tools.ToolSqlXml;

import jxl.Workbook;
import jxl.format.Alignment;
import jxl.format.Border;
import jxl.format.BorderLineStyle;
import jxl.format.Colour;
import jxl.format.VerticalAlignment;
import jxl.write.Label;
import jxl.write.WritableCell;
import jxl.write.WritableCellFormat;
import jxl.write.WritableFont;
import jxl.write.WritableSheet;
import jxl.write.WritableWorkbook;
import jxl.write.WriteException;

public class MeasureStatisticXLS extends Render {

	private static final String CONTENT_TYPE = "application/msexcel;charset=UTF-8";

	private final List<Record> data;
	private final String searchCriteria;
	private final String fileName;

	public MeasureStatisticXLS(List<Record> data,String searchCriteria, String fileName) {
		this.data = data;
		this.searchCriteria = searchCriteria;
		this.fileName = fileName;
	}

	@Override
	public void render() {
		this.response.reset();
		try {
			this.response.setHeader("Content-disposition",
					"attachment; filename*=UTF-8''" + URLEncoder.encode(this.fileName, "UTF-8") + ".xls");
		} catch (UnsupportedEncodingException e) {
			e.printStackTrace();
		}
		this.response.setContentType(CONTENT_TYPE);

		OutputStream os = null;
		try {
			os = this.response.getOutputStream();
			WritableWorkbook workbook = Workbook.createWorkbook(os);
			WritableSheet sheet = workbook.createSheet(this.fileName, 0);
			//主表头
			List<String> main_header = new ArrayList<>();
			main_header.add(fileName);
			for(int mhi = 0; mhi < 13; mhi++){
				main_header.add("");
			}
			//搜索表头
			List<String> search_header = new ArrayList<>();
			search_header.add("搜索条件：");
			search_header.add(searchCriteria);
			for(int mhi = 0; mhi < 12; mhi++){
				search_header.add("");
			}
			//第一行表头
			List<String> one_header = new ArrayList<>();
			one_header.add("所属机构");
			one_header.add("编号");
			one_header.add("姓名");
			
			one_header.add("性别");
			one_header.add("年龄");
			one_header.add("手机号");
			
			one_header.add("身高");
			one_header.add("体重");
			one_header.add("接待人");
			
			one_header.add("介绍人");
			one_header.add("进店方式");
			one_header.add("顾客分类");
			
			one_header.add("顾客反应");
			one_header.add("日期");
			
			//列名
			List<String> columns = new ArrayList<>();
			columns.add("oname");
			columns.add("number");
			columns.add("name");
			
			columns.add("sex");
			columns.add("age");
			columns.add("phone");
			
			columns.add("height");
			columns.add("weight");
			columns.add("jdrname");
		
			columns.add("jsrname");
			columns.add("jdfs");
			columns.add("gkfl");

			columns.add("gkfy");
			columns.add("gxsj");

			int row = 0;//行数

			for(int i =0;i<12;i++){
				sheet.setColumnView(i, 15); // 设置列的宽度
			}
			for(int i =0;i<=data.size()+2;i++){
				sheet.setRowView(i, 600); // 设置行的高度
			}
			sheet.setColumnView(1, 20); // 设置列的宽度
			sheet.setColumnView(13, 22); // 设置列的宽度
			
			sheet.mergeCells(0, 0, 14, 0);// 列，行，列，行
			sheet.mergeCells(1, 1, 13, 1);// 列，行，列，行
			//主表头样式
			WritableFont wf = new WritableFont(WritableFont.TIMES, 18,  WritableFont.BOLD, true);
			WritableCellFormat wcfF = new WritableCellFormat(wf);
			wcfF.setBackground(Colour.GRAY_25);// 背景色
			wcfF.setAlignment(Alignment.CENTRE);// 水平对齐
			wcfF.setVerticalAlignment(VerticalAlignment.CENTRE);// 垂直对齐
			wcfF.setBorder(Border.ALL, BorderLineStyle.THIN);// 边框
			//搜索表头样式
			WritableCellFormat search_format = new WritableCellFormat();
			search_format.setBackground(Colour.GRAY_25);// 背景色
			search_format.setVerticalAlignment(VerticalAlignment.CENTRE);// 垂直对齐
			search_format.setBorder(Border.ALL, BorderLineStyle.THIN);// 边框
			//表头样式
			WritableCellFormat header_format = new WritableCellFormat();
			header_format.setBackground(Colour.GRAY_25);// 背景色
			header_format.setAlignment(Alignment.CENTRE);// 水平对齐
			header_format.setVerticalAlignment(VerticalAlignment.CENTRE);// 垂直对齐
			header_format.setBorder(Border.ALL, BorderLineStyle.THIN);// 边框
			header_format.setWrap(true);// 是否换行
			//生成主表头
			for (int mi = 0; mi < main_header.size(); mi++) {
				String hea = main_header.get(mi);
				sheet.addCell(new Label(mi, row, hea, wcfF));
			}
			row++;
			//生成搜索表头
			for (int si = 0; si < search_header.size(); si++) {
				String hea = search_header.get(si);
				if(si==0)
				{
					sheet.addCell(new Label(si, row, hea, header_format));
				}else
				{
					sheet.addCell(new Label(si, row, hea, search_format));
				}
				
			}
			row++;
			//生成第一行表头
			for (int i = 0; i < one_header.size(); i++) {
				String header = one_header.get(i);
				sheet.addCell(new Label(i, row, header, header_format));
			}

			row++;
			
			//表格样式
			WritableCellFormat row_format = new WritableCellFormat();
			row_format.setAlignment(Alignment.CENTRE);// 水平对齐
			row_format.setVerticalAlignment(VerticalAlignment.CENTRE);// 垂直对齐
			row_format.setBorder(Border.ALL, BorderLineStyle.THIN);// 边框
			
			//生成表
			for (int j = 0; j < data.size(); j++) {
				Record record = (Record) data.get(j);

				String[] values = getValues(columns, record);
				
				for (int i = 0; i < values.length; i++) {
					sheet.addCell(new Label(i, row, values[i], row_format));
				}
				row++;
			}

			workbook.write();
			workbook.close();
		} catch (IOException e) {
			throw new RenderException(e);
		} catch (WriteException e) {
			e.printStackTrace();
		} finally {
			try {
				if (os != null) {
					os.flush();
					os.close();
				}
			} catch (IOException e) {
				LogKit.error(e.getMessage(), e);
			}
		}
	}

	private static String[] getValues(List<String> columns, Record record) {
		String[] values = new String[columns.size()];
		int i = -1;
		for (String col : columns) {
			i++;
			Object value = record.get(col);
			if (value == null) {
				continue;
			}
			values[i] = value.toString();
		}
		return values;
	}

}
